How to plot loss curves with Matplotlib